/*
 * Copyright (c) 2022, yifengling0. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. No personal names or organizations' names associated with the
 *    Atomthreads project may be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE ATOMTHREADS PROJECT AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include "atom.h"
#define ATOM_NONE_STANDARDIO
#include "drv_i2c.h"
#undef OK
#include <Arduino.h>
#include <Wire.h>
#include "debug.h"
#include "syslog.h"

static size_t i2c_bit_xfer(I2c* bus, struct i2c_msg msgs[], uint32_t num)
{
    struct i2c_msg* msg;
    Esp32I2c * i2c = (Esp32I2c * )bus;
    int ret = 0;

    TwoWire *wire = (TwoWire *)i2c->Protected.wire;
    
    for (int i = 0; i < num; i++)
    {
        int cnt = 0;
        msg = &msgs[i];

        if (msg->flags & I2C_RD)
        {
            wire->requestFrom(msg->addr, (uint8_t)msg->len);
            cnt = wire->readBytes(msg->buf, msg->len);
            if( cnt < msg->len){
                ret = -ATOM_ERROR; 
                return ret;
            }else{
                ret++;
            }
        }
        else
        {
            if( !(msg->flags & I2C_NO_START) ){
                wire->beginTransmission(msg->addr);
            }

            cnt = wire->write(msg->buf, msg->len);
            if( cnt < msg->len){
                ret = -ATOM_ERROR; 
                wire->endTransmission();
                return ret;
            }

            if( !(msg->flags & I2C_NO_START) || (i == num - 1)){
                if( wire->endTransmission() != 0){
                    ret = -ATOM_ERROR;
                }else{
                    ret++;
                }
            }else{
                ret++;
            }
        }
    }

    return ret;
}



static I2cOps _ops = {
    i2c_bit_xfer,
    NULL
};

static Esp32I2c _i2c;

void Esp32I2cCreate()
{
    _i2c.super.Protected.ops = &_ops;
    _i2c.Protected.wire = (POINTER)&Wire;
    Wire.begin(SDA, SCL);
    I2cCreate(&_i2c.super, "i2c1");
    SYSLOG(LOG_INFO, "i2c module initlized");
}